# """
# 测试数据集加载
from torch.utils.data import DataLoader
from augmentation.data_augment import DataAugment
from dataset.read_yolo_dataset import ReadYolo
from utils.collate import colle

data_augment = DataAugment()
dataset = ReadYolo(trans=data_augment)
# dataset = ReadYolo() 数据格式不满足
# collate 对照校对
data = iter(DataLoader(dataset, batch_size=4,
                       drop_last=False, collate_fn=colle))
pics, labels = next(data)
print(pics.shape)
print(labels)
# """